package com.huangd.platform.modules.sys.controller;

import com.huangd.platform.modules.sys.entity.SysRoleMenuEntity;
import com.huangd.platform.modules.sys.entity.TblRoleEntity;
import com.huangd.platform.modules.sys.entity.to.SetUserRoleTo;
import com.huangd.platform.modules.sys.service.SysRoleMenuService;
import com.huangd.platform.modules.sys.service.TblRoleService;
import com.huangd.platform.modules.sys.service.TblUserRoleService;
import com.huangd.platform.common.annotation.SysLog;
import com.huangd.platform.common.utils.Constant;
import com.huangd.platform.common.utils.PageUtils;
import com.huangd.platform.common.utils.R;
import com.huangd.platform.common.validator.ValidatorUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 角色管理
 *
 * @author huangd
 */
@RestController
@Api(tags={"用户角色相关"})
@RequestMapping("/sys/role")
public class SysRoleController extends AbstractController {
	@Resource
	private TblRoleService sysRoleService;
	@Resource
	private TblUserRoleService userRoleService;
	@Resource
	private SysRoleMenuService sysRoleMenuService;

	/**
	 * 角色列表
	 */
	@GetMapping("/list")
	@ApiOperation(value = "角色列表(带参数)")
	@RequiresPermissions("sys:role:list")
	public R list(@RequestParam Map<String, Object> params){
		//如果不是超级管理员，则只查询自己创建的角色列表
		List<Integer> list = userRoleService.queryRoleIdList(getUserId().longValue());
		if(!list.contains(Constant.SUPER_ADMIN)){
			return R.error("非管理员不能查看角色列表");
		}

		PageUtils page = sysRoleService.queryPage(params);

		return R.ok(page);
	}

	/**
	 * 角色列表
	 */
	@GetMapping("/select")
	@ApiOperation(value = "角色列表")
	@RequiresPermissions("sys:role:select")
	public R select(){
		Map<String, Object> map = new HashMap<>();
		List<Integer> list = userRoleService.queryRoleIdList(getUserId().longValue());
		if(!list.contains(Constant.SUPER_ADMIN)){
			map.put("user_id", getUserId());
		}
		List<TblRoleEntity> listData = sysRoleService.getListByMap(map);
		
		return R.ok(listData);
	}
	
	/**
	 * 角色信息
	 */
	@GetMapping("/info/{roleId}")
	@ApiOperation(value = "角色信息")
	@RequiresPermissions("sys:role:info")
	public R info(@PathVariable("roleId") Long roleId){
		TblRoleEntity role = sysRoleService.getById(roleId);
		
		//查询角色对应的菜单
		List<Long> menuIdList = sysRoleMenuService.queryMenuIdList(roleId);
		role.setMenuIdList(menuIdList);
		
		return R.ok(role);
	}
	
	/**
	 * 保存角色
	 */
	@SysLog("保存角色")
	@PostMapping("/save")
	@ApiOperation(value = "保存角色")
	@RequiresPermissions("sys:role:save")
	public R save(@RequestBody TblRoleEntity role){
		ValidatorUtils.validateEntity(role);

		sysRoleService.saveRole(role);
		
		return R.ok();
	}
	
	/**
	 * 修改角色
	 */
	@SysLog("修改角色")
	@PostMapping("/update")
	@ApiOperation(value = "修改角色")
	@RequiresPermissions("sys:role:update")
	public R update(@RequestBody TblRoleEntity role){
		ValidatorUtils.validateEntity(role);
		
		sysRoleService.update(role);
		
		return R.ok();
	}
	
	/**
	 * 删除角色
	 */
	@SysLog("删除角色")
	@PostMapping("/delete")
	@ApiOperation(value = "删除角色")
	@RequiresPermissions("sys:role:delete")
	public R delete(@RequestBody Long[] roleIds){
		sysRoleService.deleteBatch(roleIds);
		
		return R.ok();
	}
	/**
	 * 给用户分配角色
	 */
	@PostMapping("/assignRoles")
	@ApiOperation(value = "给用户分配角色")
	@RequiresPermissions("sys:role:assignRoles")
	public R assign_roles(@RequestBody SetUserRoleTo setUserRoleTo){

		userRoleService.saveRoles(setUserRoleTo);

		return R.ok();
	}
	/**
	 * 给角色分配菜单
	 */
	@PostMapping("/assignMenu")
	@ApiOperation(value = "给角色分配菜单")
	@RequiresPermissions("sys:role:assignMenu")
	public R assignMenu(@RequestBody SysRoleMenuEntity sysRoleMenuEntity){

		sysRoleMenuService.save(sysRoleMenuEntity);

		return R.ok();
	}
}
